const classify = prompt.define({
  slug: "classify-ticket",
  model: "gpt-4o-mini",
  input: (s) =>
    s.object({
      ticket: s.string(),
    }),
  output: (s) =>
    s.object({
      label: s.enum(["bug", "question"]),
    }),
  render: ({ input }) => [
    prompt.system`Classify support tickets.`,
    prompt.user`Ticket: ${input.ticket}`,
  ],
});

const messagesPrompt = prompt.define({
  slug: "reply",
  input: (s) => s.object({ ticket: s.string() }),
  render: ({ input }) => [prompt.user`Reply to: ${input.ticket}`], // <- returns array
});

const stringPrompt = prompt.define({
  slug: "policy",
  input: (s) => s.object({ policy: s.string() }),
  render: ({ input }) => prompt.text`Policy: ${input.policy}`, // <- returns tagged string
});

const brandVoicePromptDefinition = prompt.define({
  slug: "brand-voice",
  input: (s) => s.object({ company: s.string(), tone: s.string() }),
  render: ({ input }) => [
    prompt.system`Use ${input.company}'s ${input.tone} voice.`,
  ],
});

const policyTextPromptDefinition = prompt.define({
  slug: "policy-text",
  input: (s) => s.object({ company: s.string(), policy: s.string() }),
  render: ({ input }) => prompt.text`${input.company}: ${input.policy}`,
});

const replyPromptDefinition = prompt.define({
  slug: "reply",
  input: (s) =>
    s.object({
      company: s.string(),
      ticket: s.string(),
      voice: s.messagesPromptDefinition(brandVoicePromptDefinition),
      policy: s.stringPromptDefinition(policyTextPromptDefinition),
    }),
  render: ({ input }) => [
    ...input.voice,
    prompt.system`Policy: ${input.policy}`,
    prompt.user`Draft a reply for: ${input.ticket}`,
  ],
});

replyPromptDefinition.build({
  company: "Braintrust",
  ticket: "The app crashes.",
  voice: { tone: "direct" },
  policy: { policy: "Keep it short." },
});

const replyPromptDefinitionWithBuiltParts = prompt.define({
  slug: "reply-with-built-parts",
  input: (s) =>
    s.object({
      ticket: s.string(),
      voice: s.builtMessagesPrompt(),
      policy: s.builtStringPrompt(),
    }),
  render: ({ input }) => [
    ...input.voice,
    prompt.system`Policy: ${input.policy}`,
    prompt.user`Draft a reply for: ${input.ticket}`,
  ],
});

replyPromptDefinitionWithBuiltParts.build({
  ticket: "The app crashes.",
  voice: brandVoicePromptDefinition.build({ company: "Braintrust", tone: "direct" }),
  policy: policyTextPromptDefinition.build({
    company: "Braintrust",
    policy: "Keep it short.",
  }),
});

const built = promptDefinition.build({ ticket: "The app crashes." });

const args = built.to(prompt.adapters.openAIChat());

const client = new OpenAI();
const completion = await client.chat.completions.create(
  built
    .to(prompt.adapters.openAIChat())
    .extend({
      temperature: 0.2,
      tools: {
        foobar: 'baz'
      }
    })
    .extend({
      doubleExtension: 'possible'
    })
  );

const project = braintrust.projects.create({
  name: "Example",
});

project.prompts.create(promptDefinition) // <- returns void/Promise<void> not sure yet